import logging
import random

from django.http import HttpResponse
from django_redis import get_redis_connection
from rest_framework.response import Response
from rest_framework.views import APIView

from meiduo_mall.libs.captcha.captcha import captcha
from . import constants
from .serializers import CheckImageCodeSerializer

# 获取日志器
logger = logging.getLogger('django')


# GET /sms_codes/(?P<mobile>1[3-9]\d{9})/?image_code_id=xxx&text=xxx
class SMSCodeView(APIView):
    def get(self, request, mobile):
        """
        发送短信验证码:
        1. 接收参数并进行参数校验(参数合法性校验，图片验证码对比)
        2. 使用云通讯发送短信验证码
        3. 返回应答，发送成功
        """
        # 1. 接收参数并进行参数校验(参数合法性校验，图片验证码对比)
        serializer = CheckImageCodeSerializer(data=request.query_params, context={'view': self})
        serializer.is_valid(raise_exception=True)

        # 2.在redis中保存验证码内容，以键值对形式保存
        sms_code = '%06d' % random.randint(0, 999999)
        logger.info('短信验证码为: %s' % sms_code)

        redis_con = get_redis_connection('verify_codes')
        pipeline = redis_con.pipeline()
        # redis_con.setex('sms_%s' % mobile, constants.SMS_CODE_REDIS_EXPIRES, sms_code)
        # redis_con.setex('send_flag_%s' % mobile, constants.SEND_SMS_CODE_INTERVAL, 1)
        pipeline.setex('sms_%s' % mobile, constants.SMS_CODE_REDIS_EXPIRES, sms_code)
        pipeline.setex('send_flag_%s' % mobile, constants.SEND_SMS_CODE_INTERVAL, 1)
        # 一次执行管道的所有命令
        pipeline.execute()

        # try:
        #     expires = constants.SMS_CODE_REDIS_EXPIRES // 60
        #     res = CCP().send_template_sms(mobile, [sms_code, expires], constants.SMS_CODE_TEMP_ID)
        # except Exception as e:
        #     logger.error(e)
        #     return Response({'message': '发送短信异常'}, status=status.HTTP_503_SERVICE_UNAVAILABLE)
        # else:
        #     if res != 0:
        #         # 发送短信失败
        #         logger.error('发送短信失败')
        #         return Response({'message': '发送短信失败'}, status=status.HTTP_503_SERVICE_UNAVAILABLE)

        # 使用celery发出发送短信的任务
        from celery_tasks.sms.tasks import send_sms_code
        expires = constants.SMS_CODE_REDIS_EXPIRES // 60
        send_sms_code.delay(mobile, sms_code, expires)

        # 3. 返回应答，发送成功
        return Response({'message': '发送短信成功'})


# Create your views here.
# url('^image_codes/(?P<image_code_id>[\w-]+)/$', views.ImageCodeView.as_view()),
class ImageCodeView(APIView):
    """
    图片验证码
    """

    def get(self, request, image_code_id):
        """产生图片验证码:"""
        # 1. 产生图片验证码captcha()
        # 这个是通过第三方包生成的，生产验证码的文本和图片，有兴趣的可以自己去源代码中看一下
        text, image = captcha.generate_captcha()
        logger.info('图片验证码%s' % text)

        # 2.在redis中保存图片验证码
        # get_reids_connection('redis中的名字'),根据参数链接到redis中的对象，需要在配置文件中提前配置这个对象
        redis_con = get_redis_connection('verify_codes')
        # redis_con.set('key', 'val', 'expires'),　这个函数能够设置redis对象的键值对，并且设定保存时间
        # 创建constants.py文件，写入验证码的缓存时间
        redis_con.setex('img_%s' % image_code_id, constants.IMAGE_CODE_REDIS_EXPIRES, text)

        # 3.返回图片验证码
        return HttpResponse(image, content_type='image/jpg')
